package ModelLayer;

import java.util.ArrayList;

public class TreatmentCategory {
	
	private int id;
	private String name;
	private ArrayList<Treatment> treatments;
	
	public TreatmentCategory()
	{
		
	}
	
	public TreatmentCategory(String name)
	{
		this.name = name;
	}
	
	public TreatmentCategory(int id, String name)
	{
		this.id = id;
		this.name = name;
		treatments = new ArrayList<Treatment>();
	}
	
	public TreatmentCategory(int id)
	{
		this.id = id;
	}

	public int getId() 
	{
		return id;
	}

	public void setId(int id) 
	{
		this.id = id;
	}

	public String getName() 
	{
		return name;
	}

	public void setName(String name) 
	{
		this.name = name;
	}
	
	public void addTreatments(Treatment treat)
	{
		treatments.add(treat);
	}

	public ArrayList<Treatment> getTreatments() 
	{
		return treatments;
	}

	public void setTreatments(ArrayList<Treatment> treatments) 
	{
		this.treatments = treatments;
	}
	
	public String toString()
	{
		return name;
	}
}
